import os,sys
import os.path as osp
import pandas as pd

def output_target_cate_metric():
    "从csv里面输出指定类的相关指标"
    csv_path=r"/Users/edy/Desktop/yolov5/runs/train/db378_132/db378_132_metrics.csv"
    target_category=[
'依仑大白梨汽水500ml',
'依仑大连汽水500ml',
'依仑大连汽水桔子味500ml',
'依仑大连汽水苹果味500ml',
'依仑大连荔枝味碳酸饮料500ml',
'依仑小蜜蜂葡萄果味型碳酸饮料500ml',
'八王寺山楂香梨维生素果汁汽水380ml',
'八王寺荔枝维生素果汁汽水380ml',
'八王寺香橙维生素果汁汽水380ml',
'大窑嘉宾果味汽水520ml',
'大窑嘉宾果味汽水PET520ml',
'大窑橙诺520ml',
'大窑橙诺PET520ml',
'宏宝莱果汁鲜山楂味饮料玻璃瓶200ml',
'宏宝莱果汁鲜桔子味汽水饮料玻璃瓶200ml',
'宏宝莱果汁鲜脐橙味饮料玻璃瓶200ml',
'宏宝莱生榨山水蜜桃果味饮料300ml',
'宏宝莱生榨蓝莓苹果复合果味饮料300ml',
'宏宝莱生榨香橙果肉饮料300ml',
'宏宝莱老汽水550ml',
'宏宝莱花生露植物蛋白饮料445ml',
'宏宝莱花生露植物蛋白饮料玻璃瓶200ml',
'宏宝莱荔枝味汽水饮料玻璃瓶200ml',
'宏宝莱荔枝汽水330ml铝罐',
'旺仔大白梨果味型碳酸饮料310ml',
'珍珍桃味汽水330ml',
'珍珍桔味汽水330ml',
'珍珍荔枝味汽水330ml',
'秋林格瓦斯PE瓶350ml',
'秋林黑格瓦斯PE瓶350ml',
    ]
    
    df=pd.read_csv(csv_path)
    precision=df['precision']
    recall=df['recall']
    right_rate=df['right_rate']
    ap=df['ap']
    category_name=df['Unnamed: 0']
    
    for ind in range(len(category_name)):
        if category_name[ind] in target_category:
            print("%s : right_rate  %f,    recall  %f,  ap  %f, precision  %f"%(category_name[ind],right_rate[ind],recall[ind],ap[ind],precision[ind],))
    
    


def main():
    output_target_cate_metric()

if __name__ == '__main__':
    main()